class FeatureDO:
    def __init__(
        self,
        id: int,
        md5: str,
        famLabel: str,
        graphEmbedding: str,
        isTrainSample: int,
    ):
        self.id = id
        self.md5 = md5
        self.famLabel = famLabel
        self.graphEmbedding = graphEmbedding
        self.isTrainSample = isTrainSample

    def DO2Dict(self):
        return {
            "id": self.id,
            "md5": self.md5,
            "fam_label": self.famLabel,
            "graph_embedding": self.graphEmbedding,
            "is_train_sample": self.isTrainSample,
        }
